17905f56c9c697dd3b0c34cfbc184adb3d643af8,modules/assignment/src/test/java/com/intuit/wasabi/assignment/impl/AssignmentsImplTest.java,AssignmentsImplTest,getAssignment_test_4,#,840

Before Change


        Mockito.when(assignmentsRepository.assignUser(assignment, experiment, DATE)).thenReturn(newAssignment);
        Mockito.when(assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-b"), context)).thenReturn(newAssignment);

        assignment = assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-b"), context);
        assertTrue(assignment.getBucketLabel().equals(experienceB.getLabel()));
        assertTrue(assignment.getStatus() == Assignment.Status.NEW_ASSIGNMENT);

        newAssignment = Assignment.newInstance(assignment.getExperimentID())
                .withBucketLabel(assignment.getBucketLabel())
                .withUserID(assignment.getUserID())
                .withContext(assignment.getContext())
                .withStatus(Assignment.Status.EXISTING_ASSIGNMENT)
                .withCreated(DATE)
                .withCacheable(null)
                .build();

        Mockito.when(assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-b"), context)).thenReturn(newAssignment);
        assignment = assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-b"), context);

        assertTrue(assignment.getStatus() == Assignment.Status.EXISTING_ASSIGNMENT);
        assertTrue(assignment.getBucketLabel().equals(experienceB.getLabel()));
        assertTrue(assignment.getStatus() == Assignment.Status.EXISTING_ASSIGNMENT);

        builder = Assignment.newInstance(experiment.getID())
                .withApplicationName(experiment.getApplicationName())
                .withUserID(User.ID.valueOf("user-c"))
                .withContext(context)
                .withBucketLabel(experienceC.getLabel());
        assignment = builder.build();

        newAssignment = Assignment.newInstance(assignment.getExperimentID())
                .withBucketLabel(assignment.getBucketLabel())
                .withUserID(assignment.getUserID())
                .withContext(assignment.getContext())
                .withStatus(Assignment.Status.NEW_ASSIGNMENT)
                .withCreated(DATE)
                .withCacheable(null)
                .build();
        Mockito.when(assignmentsRepository.assignUser(assignment, experiment, DATE)).thenReturn(newAssignment);
        Mockito.when(assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-c"), context)).thenReturn(newAssignment);
        assignment = assignmentsRepository.getAssignment(experiment.getID(), User.ID.valueOf("user-c"), context);

        assertTrue(assignment.getBucketLabel() == experienceC.getLabel());
        assertTrue(assignment.getStatus() == Assignment.Status.NEW_ASSIGNMENT);

After Change


                .withCacheable(null)
                .build();

        Mockito.when(assignmentsRepository.getAssignment(User.ID.valueOf("user-b"), testApp, experiment.getID(), context)).thenReturn(newAssignment);
        assignment = assignmentsRepository.getAssignment(User.ID.valueOf("user-b"), testApp, experiment.getID(), context);

        assertTrue(assignment.getStatus() == Assignment.Status.EXISTING_ASSIGNMENT);
        assertTrue(assignment.getBucketLabel().equals(experienceB.getLabel()));
        assertTrue(assignment.getStatus() == Assignment.Status.EXISTING_ASSIGNMENT);

        builder = Assignment.newInstance(experiment.getID())
                .withApplicationName(experiment.getApplicationName())
                .withUserID(User.ID.valueOf("user-c"))
                .withContext(context)
                .withBucketLabel(experienceC.getLabel());
        assignment = builder.build();

        newAssignment = Assignment.newInstance(assignment.getExperimentID())
                .withBucketLabel(assignment.getBucketLabel())
                .withUserID(assignment.getUserID())
                .withContext(assignment.getContext())
                .withStatus(Assignment.Status.NEW_ASSIGNMENT)
                .withCreated(DATE)
                .withCacheable(null)
                .build();
        Mockito.when(assignmentsRepository.getAssignment(User.ID.valueOf("user-c"), testApp, experiment.getID(), context)).thenReturn(newAssignment);
        assignment = assignmentsRepository.getAssignment(User.ID.valueOf("user-c"), testApp, experiment.getID(), context);

        assertTrue(assignment.getBucketLabel() == experienceC.getLabel());